// Copyright (c) Appeon. All rights reserved.
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. 
// You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0.

using System;
using Microsoft.Extensions.DependencyInjection;

namespace SnapObjects.Data
{
    public static class DataExtensions
    {
        public static IServiceCollection AddDataContext<TAdoDbContext>(
            this IServiceCollection services,
            Func<IDataContextOptions<TAdoDbContext>, IDataContextOptions<TAdoDbContext>> contextOptions,
            ServiceLifetime contextLifetime = ServiceLifetime.Scoped)
            where TAdoDbContext : class
        {
            IDataContextOptions<TAdoDbContext> options = null;

            options = contextOptions(options);

            services.AddSingleton(options);

            switch (contextLifetime)
            {
                case ServiceLifetime.Scoped:
                    services.AddScoped<TAdoDbContext>();
                    break;
                case ServiceLifetime.Singleton:
                    services.AddSingleton<TAdoDbContext>();
                    break;
                case ServiceLifetime.Transient:
                    services.AddTransient<TAdoDbContext>();
                    break;
            }

            return services;
        }
    }
}
